package com.leecode.xiehf.page_02;

import com.leecode.Printer;

/**
 * 给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。
 *
 * 最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
 *
 * 你可以假设除了整数 0 之外，这个整数不会以零开头。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/plus-one
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。。
 */
public class Solution_0066 extends Printer {

    public static void main(String[] args) {
        Solution_0066 solution = new Solution_0066();
        int[] s = solution.plusOne(new int[]{
                9, 9, 9
        });
        print(s);
    }

    public int[] plusOne(int[] digits) {
        int up = 1;
        for (int i = digits.length - 1; i >= 0; i--) {
            int digit = digits[i];
            int result = digit + up;
            digits[i] = result % 10;
            up = result / 10;
        }
        if (up > 0) {
            int[] re = new int[digits.length + 1];
            re[0] = 1;
            for (int i = 1; i < re.length; i++) {
                re[i] = digits[i - 1];
            }
            return re;
        } else {
            return digits;
        }
    }
}
